python for quadratic equation